Python Error: ValueError: shapes (4 2) and (4 2) not aligned: 2 (dim 1) != 4 (dim 0)